package com.cnpc.framework.utils;

import org.apache.poi.hssf.usermodel.*;
import org.apache.poi.poifs.filesystem.POIFSFileSystem;
import org.apache.poi.ss.usermodel.Cell;
import org.apache.poi.ss.usermodel.CellStyle;
import org.apache.poi.ss.usermodel.Row;
import org.apache.poi.ss.usermodel.Sheet;
import org.apache.poi.ss.util.CellRangeAddressList;
import org.apache.poi.xssf.usermodel.XSSFWorkbook;

import javax.servlet.http.HttpServletRequest;
import java.io.File;
import java.io.FileInputStream;
import java.io.InputStream;
import java.util.List;

public class ExcelExportUtil {

    /**
     * 导出Excel
     *
     * @param sheetName sheet名称
     * @param title     标题
     * @param values    内容
     * @param wb        HSSFWorkbook对象
     * @return
     */
    public static HSSFWorkbook getHSSFWorkbook(String sheetName, String[] title, String[][] values, HSSFWorkbook wb) {

        // 第一步，创建一个HSSFWorkbook，对应一个Excel文件
        if (wb == null) {
            wb = new HSSFWorkbook();
        }

        // 第二步，在workbook中添加一个sheet,对应Excel文件中的sheet
        HSSFSheet sheet = wb.createSheet(sheetName);

        // 第三步，在sheet中添加表头第0行,注意老版本poi对Excel的行数列数有限制
        HSSFRow row = sheet.createRow(0);

        // 第四步，创建单元格，并设置值表头 设置表头居中
        HSSFCellStyle style = wb.createCellStyle();
        style.setAlignment(HSSFCellStyle.ALIGN_CENTER); // 创建一个居中格式

        //数据校验
        //声明列对象
        HSSFCell cell = null;

        //创建标题
        for (int i = 0; i < title.length; i++) {
            cell = row.createCell(i);
            cell.setCellValue(title[i]);
            cell.setCellStyle(style);

        }

        //创建内容
        for (int i = 0; i < values.length; i++) {
            row = sheet.createRow(i + 1);
            for (int j = 0; j < values[i].length; j++) {
                //将内容按顺序赋给对应的列对象
                cell = row.createCell(j);
                cell.setCellValue(values[i][j]);
                cell.setCellType(Cell.CELL_TYPE_STRING);
            }
        }
        return wb;
    }


    /**
     * 导出Excel
     *
     * @param tempName 模板路径
     * @param rowStart
     * @param values   内容
     * @return
     */
    public static HSSFWorkbook getHSSFWorkbookByTemplate(String tempName, int rowStart, String[][] values, HttpServletRequest request) throws Exception {
        String basePath = request.getSession().getServletContext().getRealPath("/");
        String filePath = "/uploadPath/excelmodel/" + tempName;
        File file = new File(basePath + filePath);

        if (!file.exists()) {
            return null;
        }
        InputStream inp = new FileInputStream(file);
        POIFSFileSystem fs = new POIFSFileSystem(inp);
        HSSFWorkbook wb = new HSSFWorkbook(fs);


        // 第二步，在workbook中添加一个sheet,对应Excel文件中的sheet
        HSSFSheet sheet =wb.getSheetAt(0);

        // 第三步，在sheet中添加表头第0行,注意老版本poi对Excel的行数列数有限制
        HSSFRow row = null;

        // 第四步，创建单元格，并设置值表头 设置表头居中
        HSSFCellStyle style = wb.createCellStyle();
        style.setAlignment(HSSFCellStyle.ALIGN_CENTER); // 创建一个居中格式

        //声明列对象
        HSSFCell cell = null;

        //创建内容
        for (int i = 0; i < values.length; i++) {
            row = sheet.createRow(rowStart-1+i);
            for (int j = 0; j < values[i].length; j++) {
                //将内容按顺序赋给对应的列对象
                cell = row.createCell(j);
                cell.setCellValue(values[i][j]);
                cell.setCellType(Cell.CELL_TYPE_STRING);
            }
        }
        return wb;
    }

    /**
     * 导出Excel
     *
     * @param tempName 模板路径
     * @param rowStart
     * @param values   内容
     * @return
     */
    public static XSSFWorkbook getXSSFWorkbookByTemplate(String tempName,int sheetIndex, int rowStart, String[][] values, HttpServletRequest request) throws Exception {
        String basePath = request.getSession().getServletContext().getRealPath("/");
        String filePath = "/uploadPath/excelmodel/" + tempName;
        File file = new File(basePath + filePath);

        if (!file.exists()) {
            return null;
        }
        InputStream inp = new FileInputStream(file);
        XSSFWorkbook wb = new XSSFWorkbook(inp);


        // 第二步，在workbook中添加一个sheet,对应Excel文件中的sheet
        Sheet sheet =wb.getSheetAt(sheetIndex);

        Row row = null;

        // 第四步，创建单元格，并设置值表头 设置表头居中
        CellStyle style = wb.createCellStyle();
        style.setAlignment(HSSFCellStyle.ALIGN_CENTER); // 创建一个居中格式

        //声明列对象
        Cell cell = null;

        //创建内容
        for (int i = 0; i < values.length; i++) {
            row = sheet.createRow(rowStart-1+i);
            for (int j = 0; j < values[i].length; j++) {
                System.out.println("value:"+values[i][j]);
                //将内容按顺序赋给对应的列对象
                cell = row.createCell(j);
                cell.setCellValue(values[i][j]);
                cell.setCellType(Cell.CELL_TYPE_STRING);
            }
        }
        return wb;
    }

    /**
     * 导出Excel
     *
     * @param sheetName sheet名称
     * @param title     标题
     * @param values    内容
     * @param wb        HSSFWorkbook对象
     * @return
     */
    public static HSSFWorkbook getHSSFWorkbook(String sheetName, List<String> title, List<List<String>> values, HSSFWorkbook wb) {

        // 第一步，创建一个HSSFWorkbook，对应一个Excel文件
        if (wb == null) {
            wb = new HSSFWorkbook();
        }

        // 第二步，在workbook中添加一个sheet,对应Excel文件中的sheet
        HSSFSheet sheet = wb.createSheet(sheetName);

        // 第三步，在sheet中添加表头第0行,注意老版本poi对Excel的行数列数有限制
        HSSFRow row = sheet.createRow(0);

        // 第四步，创建单元格，并设置值表头 设置表头居中
        HSSFCellStyle style = wb.createCellStyle();
        style.setAlignment(HSSFCellStyle.ALIGN_CENTER); // 创建一个居中格式

        //数据校验
        //声明列对象
        HSSFCell cell = null;

        //创建标题
        for (int i = 0; i < title.size(); i++) {
            cell = row.createCell(i);
            cell.setCellValue(title.get(i));
            cell.setCellStyle(style);
        }

        //创建内容
        for (int i = 0; i < values.size(); i++) {
            row = sheet.createRow(i + 1);
            for (int j = 0; j < values.get(i).size(); j++) {
                //将内容按顺序赋给对应的列对象
                row.createCell(j).setCellValue(values.get(i).get(j));
            }
        }
        return wb;
    }

    // 下拉框限制
    public static HSSFDataValidation setBoxs(String[] str, Integer col) {
        CellRangeAddressList addressList = new CellRangeAddressList(1, 1000, col, col);
        //  final String[] DATA_LIST = new String[] { "男", "女", };
        DVConstraint dvConstraint = DVConstraint.createExplicitListConstraint(str);

        HSSFDataValidation dataValidation = new HSSFDataValidation(addressList, dvConstraint);
        dataValidation.setSuppressDropDownArrow(false);
        //dataValidation.createPromptBox("输入提示", "请从下拉列表中选择值");
        dataValidation.setShowPromptBox(true);

        return dataValidation;
    }

}
